﻿/*
 * Copyright 2012 The Netty Project
 *
 * The Netty Project licenses this file to you under the Apache License,
 * version 2.0 (the "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at:
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 *
 * Copyright (c) The DotNetty Project (Microsoft). All rights reserved.
 *
 *   https://github.com/azure/dotnetty
 *
 * Licensed under the MIT license. See LICENSE file in the project root for full license information.
 *
 * Copyright (c) 2020 The Dotnetty-Span-Fork Project (cuteant@outlook.com) All rights reserved.
 *
 *   https://github.com/cuteant/dotnetty-span-fork
 *
 * Licensed under the MIT license. See LICENSE file in the project root for full license information.
 */

namespace DotNetty.Transport.Channels.Sockets
{
    using System;
    using System.Net.Sockets;
    using DotNetty.Common.Utilities;

    public class SocketChannelAsyncOperation<TChannel, TUnsafe> : SocketAsyncEventArgs
        where TChannel : AbstractSocketChannel<TChannel, TUnsafe>
        where TUnsafe : AbstractSocketChannel<TChannel, TUnsafe>.AbstractSocketUnsafe, new()
    {
        public SocketChannelAsyncOperation(TChannel channel)
            : this(channel, true)
        {
        }

        public SocketChannelAsyncOperation(TChannel channel, bool setEmptyBuffer)
        {
            if (channel is null) { ThrowHelper.ThrowArgumentNullException(ExceptionArgument.channel); }

            Channel = channel;
            Completed += AbstractSocketChannel<TChannel, TUnsafe>.IoCompletedCallback;
            if (setEmptyBuffer)
            {
#if NETCOREAPP || NETSTANDARD_2_0_GREATER
                SetBuffer(Memory<byte>.Empty);
#else
                SetBuffer(ArrayExtensions.ZeroBytes, 0, 0);
#endif
            }
        }

        public void Validate()
        {
            SocketError socketError = SocketError;
            if (socketError != SocketError.Success)
            {
                ThrowHelper.ThrowSocketException(socketError);
            }
        }

        public TChannel Channel { get; private set; }
    }
}